package com.smartStatus.data.utils;

import com.alibaba.excel.EasyExcel;
import com.alibaba.fastjson.JSON;
import org.apache.poi.ss.formula.functions.T;
import org.springframework.web.multipart.MultipartFile;

import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.net.URLEncoder;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * @ClassName : ExcelUtils
 * @Author : lby
 * @Date: 2021/1/18 16:47
 * @Description : easyExcel工具类
 */
public class ExcelUtils<T> {


    /**
     * 文件下载（失败了会返回一个有部分数据的Excel）
     *
     * @param dtoClass  dto类
     * @param response  httpServletResponse
     * @param list      导出数据
     * @param sheetName 模板名称
     * @throws IOException
     */
    public static <T> void exportData(Class<T> dtoClass, HttpServletResponse response, List list) throws IOException {
        // 这里注意 有同学反应使用swagger 会导致各种问题，请直接用浏览器或者用postman
        response.setContentType("application/vnd.ms-excel");
        response.setCharacterEncoding("utf-8");
        // 这里URLEncoder.encode可以防止中文乱码 当然和easyexcel没有关系
        String fileName = URLEncoder.encode("测试", "UTF-8").replaceAll("\\+", "%20");
        fileName = URLEncoder.encode("测试", "UTF-8");
        response.setHeader("Content-disposition", "attachment;filename=" + fileName + ".xlsx");
        EasyExcel.write(response.getOutputStream(), dtoClass).sheet("").doWrite(list);
    }

    /**
     * 文件下载并且失败的时候返回json（默认失败了会返回一个有部分数据的Excel）
     *
     * @param sheetName 模板名称
     * @param dtoClass  dto类
     * @param response  httpServletResponse
     * @param list      导出数据
     * @throws IOException
     */
    public static <T> void exportData01(Class<T> dtoClass, HttpServletResponse response, List list) throws IOException {
        // 这里注意 有同学反应使用swagger 会导致各种问题，请直接用浏览器或者用postman
        try {
            response.setContentType("application/vnd.ms-excel");
            response.setCharacterEncoding("utf-8");
            // 这里URLEncoder.encode可以防止中文乱码 当然和easyexcel没有关系
            String fileName = URLEncoder.encode("测试", "UTF-8").replaceAll("\\+", "%20");
            response.setHeader("Content-disposition", "attachment;filename*=utf-8''" + fileName + ".xlsx");
            // 这里需要设置不关闭流
            EasyExcel.write(response.getOutputStream(), dtoClass).autoCloseStream(Boolean.FALSE).sheet("")
                    .doWrite(list);
        } catch (Exception e) {
            // 重置response
            response.reset();
            response.setContentType("application/json");
            response.setCharacterEncoding("utf-8");
            Map<String, String> map = new HashMap<String, String>();
            map.put("status", "failure");
            map.put("message", "下载文件失败" + e.getMessage());
            response.getWriter().println(JSON.toJSONString(map));
        }
    }

    /**
     * @param file     导入文件
     * @param dtoClass 文件dto类
     * @return
     * @throws IOException
     */
    public static <T> List<T> importData(MultipartFile file, Class<T> dtoClass) throws IOException {
        return EasyExcel.read(file.getInputStream()).head(dtoClass).sheet().doReadSync();
    }
}
